4 * @covers PageDataRequestHandler
7 class PageDataRequestHandlerTest
extends \MediaWikiLangTestCase
{
12 private $interfaceTitle;
19 protected function setUp() {
22 $this->interfaceTitle
= Title
::newFromText( __CLASS__
);
23 $this->obLevel
= ob_get_level();
25 $this->setMwGlobals( 'wgArticlePath', '/wiki/$1' );
28 protected function tearDown() {
29 $obLevel = ob_get_level();
31 while ( ob_get_level() > $this->obLevel
) {
35 if ( $obLevel !== $this->obLevel
) {
36 $this->fail( "Test changed output buffer level: was {$this->obLevel}" .
37 "before test, but $obLevel after test."
45 * @return PageDataRequestHandler
47 protected function newHandler() {
48 return new PageDataRequestHandler();
52 * @param array $params
53 * @param string[] $headers
57 protected function makeOutputPage( array $params, array $headers ) {
59 $request = new FauxRequest( $params );
60 $request->response()->header( 'Status: 200 OK', true, 200 ); // init/reset
62 foreach ( $headers as $name => $value ) {
63 $request->setHeader( strtoupper( $name ), $value );
66 // construct Context and OutputPage
67 $context = new DerivativeContext( RequestContext
::getMain() );
68 $context->setRequest( $request );
70 $output = new OutputPage( $context );
71 $output->setTitle( $this->interfaceTitle
);
72 $context->setOutput( $output );
77 public function handleRequestProvider() {
80 $cases[] = [ '', [], [], 'Invalid title', 400 ];
84 [ 'target' => 'Helsinki' ],
88 [ 'Location' => '?title=Helsinki&action=raw' ]
92 foreach ( $cases as $c ) {
96 if ( isset( $case[1]['target'] ) ) {
97 $case[0] .= $case[1]['target'];
98 unset( $case[1]['target'] );
101 $subpageCases[] = $case;
104 $cases = array_merge( $cases, $subpageCases );
108 [ 'target' => 'Helsinki' ],
109 [ 'Accept' => 'text/HTML' ],
112 [ 'Location' => '/wiki/Helsinki' ]
118 'target' => 'Helsinki',
119 'revision' => '4242',
121 [ 'Accept' => 'text/HTML' ],
124 [ 'Location' => '?title=Helsinki&oldid=4242' ]
133 [ 'Location' => '?title=Helsinki&action=raw' ]
136 // #31: /Q5 with "Accept: text/foobar" triggers a 406
140 [ 'Accept' => 'text/foobar' ],
141 'No matching format found',
148 [ 'Accept' => 'text/HTML' ],
156 [ 'Accept' => 'text/HTML' ],
164 [ 'Accept' => 'text/HTML' ],
172 [ 'Accept' => 'text/HTML' ],
175 [ 'Location' => '/wiki/Helsinki' ]
181 [ 'Accept' => 'text/HTML' ],
184 [ 'Location' => '/wiki/Helsinki' ]
190 [ 'Accept' => 'text/HTML' ],
193 [ 'Location' => '/wiki/AC/DC' ]
200 * @dataProvider handleRequestProvider
202 * @param string $subpage The subpage to request (or '')
203 * @param array $params Request parameters
204 * @param array $headers Request headers
205 * @param string $expectedOutput
206 * @param int $expectedStatusCode Expected HTTP status code.
207 * @param string[] $expectedHeaders Expected HTTP response headers.
209 public function testHandleRequest(
213 $expectedOutput = '',
214 $expectedStatusCode = 200,
215 array $expectedHeaders = []
217 $output = $this->makeOutputPage( $params, $headers );
218 $request = $output->getRequest();
220 /* @var FauxResponse $response */
221 $response = $request->response();
224 $handler = $this->newHandler();
228 $handler->handleRequest( $subpage, $request, $output );
230 if ( $output->getRedirect() !== '' ) {
231 // hack to apply redirect to web response
235 $text = ob_get_clean();
237 $this->assertEquals( $expectedStatusCode, $response->getStatusCode(), 'status code' );
238 $this->assertSame( $expectedOutput, $text, 'output' );
240 foreach ( $expectedHeaders as $name => $exp ) {
241 $value = $response->getHeader( $name );
242 $this->assertNotNull( $value, "header: $name" );
243 $this->assertInternalType( 'string', $value, "header: $name" );
244 $this->assertStringEndsWith( $exp, $value, "header: $name" );
246 } catch ( HttpError
$e ) {
248 $this->assertEquals( $expectedStatusCode, $e->getStatusCode(), 'status code' );
249 $this->assertContains( $expectedOutput, $e->getHTML(), 'error output' );
252 // We always set "Access-Control-Allow-Origin: *"
253 $this->assertSame( '*', $response->getHeader( 'Access-Control-Allow-Origin' ) );
256 public function provideHttpContentNegotiation() {
257 $helsinki = Title
::newFromText( 'Helsinki' );
259 'Accept Header of HTML' => [
261 [ 'ACCEPT' => 'text/html' ], // headers
264 'Accept Header without weights' => [
266 [ 'ACCEPT' => '*/*, text/html, text/x-wiki' ],
267 'Helsinki&action=raw'
269 'Accept Header with weights' => [
271 [ 'ACCEPT' => 'text/*; q=0.5, text/json; q=0.7, application/rdf+xml; q=0.8' ],
272 'Helsinki&action=raw'
274 'Accept Header accepting evertyhing and HTML' => [
276 [ 'ACCEPT' => 'text/html, */*' ],
277 'Helsinki&action=raw'
279 'No Accept Header' => [
282 'Helsinki&action=raw'
288 * @dataProvider provideHttpContentNegotiation
290 * @param Title $title
291 * @param array $headers Request headers
292 * @param string $expectedRedirectSuffix Expected suffix of the HTTP Location header.
296 public function testHttpContentNegotiation(
299 $expectedRedirectSuffix
301 /* @var FauxResponse $response */
302 $output = $this->makeOutputPage( [], $headers );
303 $request = $output->getRequest();
305 $handler = $this->newHandler();
306 $handler->httpContentNegotiation( $request, $output, $title );
308 $this->assertStringEndsWith(
309 $expectedRedirectSuffix,
310 $output->getRedirect(),